package com.eastfair.projectcore.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.eastfair.auth.entity.Role;
import com.eastfair.constant.BusinessConstant;
import com.eastfair.model.DataIsolationVo;
import com.eastfair.projectcore.dto.*;
import com.eastfair.projectcore.entity.DataIsolation;
import com.eastfair.projectcore.entity.Dictionary;
import com.eastfair.projectcore.service.DataIsolationService;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import com.eastfair.boot.controller.SuperController;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.eastfair.echo.core.EchoService;
import com.eastfair.core.base.R;
import com.eastfair.projectcore.vo.ProjectDbVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import com.eastfair.boot.request.PageParams;
import com.eastfair.core.context.ContextUtil;


/**
 * <p>
 * 前端控制器
 * 数据隔离表
 * </p>
 *
 * @author ligang
 * @date 2022-06-01
 */
@Slf4j
@Validated
@RestController
@RequestMapping("/dataIsolation")
@Api(value = "DataIsolation", tags = "数据隔离表")
public class DataIsolationController extends SuperController<DataIsolationService, Long, DataIsolation, DataIsolationPageQuery, DataIsolationSaveDTO, DataIsolationUpdateDTO> {

    @Autowired
    private EchoService echoService;

    @Override
    public void handlerResult(IPage<DataIsolation> page) {
        // 想让返回值实现自动回显，请将此行代码打开
        // echoService.action(page);
    }

    /**
     * Excel导入后的操作
     *
     * @param list
     */
    @Override
    public R<Boolean> handlerImport(List<Map<String, String>> list) {
        List<DataIsolation> dataIsolationList = list.stream().map((map) -> {
            DataIsolation dataIsolation = DataIsolation.builder().build();
            //TODO 请在这里完成转换
            return dataIsolation;
        }).collect(Collectors.toList());

        return R.success(baseService.saveBatch(dataIsolationList));
    }


    /**
     * 处理参数
     *
     * @param params 分页参数
     */
    @Override
    public void handlerQueryParams(PageParams<DataIsolationPageQuery> params) {
        DataIsolationPageQuery pageQuery = params.getModel();
        pageQuery.setProjectId(ContextUtil.getProjectId());
        params.setModel(pageQuery);
    }


    /**
     * 查询项目的子系统集合
     *
     * @return
     */
    @PostMapping("/queryList")
    public R<List<DataIsolationVo>> queryList(@RequestBody DataIsolationDTO dataIsolationDTO) {
        List<DataIsolationVo> result = new ArrayList<>();
        QueryWrapper<DataIsolation> queryWrapper = new QueryWrapper<>();
        if (dataIsolationDTO != null) {
            if (dataIsolationDTO.getId() != null) {
                queryWrapper.lambda().eq(DataIsolation::getId, dataIsolationDTO.getId());
            }
            if (dataIsolationDTO.getSystemScene() != null) {
                queryWrapper.lambda().eq(DataIsolation::getSystemScene, dataIsolationDTO.getSystemScene());
            }
            if (dataIsolationDTO.getSystemSceneList() != null && !dataIsolationDTO.getSystemSceneList().isEmpty()) {
                queryWrapper.lambda().in(DataIsolation::getSystemScene, dataIsolationDTO.getSystemSceneList());
            }
        }
        if (ContextUtil.getProjectId() != null) {
            queryWrapper.lambda().eq(DataIsolation::getProjectId, ContextUtil.getProjectId());
        }
        queryWrapper.lambda().eq(DataIsolation::getIsDeleted, BusinessConstant.DELETE_NO);
        List<DataIsolation> dataIsolationList = baseService.list(queryWrapper);
        if (dataIsolationList != null && !dataIsolationList.isEmpty()) {
            for (DataIsolation dataIsolation : dataIsolationList) {
                DataIsolationVo dataIsolationVo = new DataIsolationVo();
                BeanUtils.copyProperties(dataIsolation, dataIsolationVo);
                dataIsolationVo.setSystemScene(dataIsolation.getSystemScene().getCode());
                dataIsolationVo.setDataScope(dataIsolation.getDataScope() == null ? "" : dataIsolation.getDataScope().getCode());
                dataIsolationVo.setId(dataIsolation.getId());
                dataIsolationVo.setBusinDataScope(dataIsolation.getBusinDataScope() == null ? "" : dataIsolation.getBusinDataScope().getCode());
                dataIsolationVo.setSystemType(dataIsolation.getSystemType() == null ? "" : dataIsolation.getSystemType().getCode());
                result.add(dataIsolationVo);
            }
        }
        return R.success(result);
    }

    /**
     * 创建子系统和初始化子系统
     * @return 查询结果
     */
    @ApiOperation(value = "创建子系统和初始化子系统", notes = "创建子系统和初始化子系统")
    @GetMapping("/createSystem")
    public R<List<DataIsolationVo>> createSystem(Long projectId) {
        List<DataIsolationVo> dataIsolationVoList = baseService.createSystem(projectId);
        return success(dataIsolationVoList);
    }

}
